# -*- coding: utf-8 -*-
# ----------------------------
# @Time    : 2025/4/3 14:52
# @Author  : acedar
# @FileName: graphrag_server.py
# ----------------------------

from mcp.server.fastmcp import FastMCP

mcp = FastMCP("graphrag_mcp")
USER_AGENT = "graphrag_mcp-app/1.0"

DATA_ROOT = "./data/graphrag"


@mcp.tool()
async def rag_query(query: str) -> str:
    """
    为斗破苍穹小说提供相关的知识补充
    :param query:
    :return:
    """
    return "demo_" + query


if __name__ == "__main__":
    # 以标准 I/O 方式运行
    mcp.run(transport='stdio')
